//在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。 
//
// 示例 1: 
//
// 输入: [3,2,1,5,6,4] 和 k = 2
//输出: 5
// 
//
// 示例 2: 
//
// 输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
//输出: 4 
//
// 说明: 
//
// 你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。 
// Related Topics 堆 分治算法 
// 👍 1114 👎 0

/**
 * @author Shayne
 */
package leetcode.editor.cn;

public class KthLargestElementInAnArray {
    public static void main(String[] args) {

        Solution solution = new KthLargestElementInAnArray().new Solution();
        int[] nums = {3,2,3,1,2,4,5,5,6,7,7,8,2,3,1,1,1,10,11,5,6,2,4,7,8,5,6};
        solution.findKthLargest(nums, 2);
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {

        /**
         * 思路： 快速排序
         * 以pivot为基准，比pivot小的在左边，比pivot大的在右边
         * 如果左边的index为nums.length - k, 此index即为所求值
         */

        public int findKthLargest(int[] nums, int k) {
            return quickSort(nums, 0, nums.length - 1, nums.length - k);
        }

        private int quickSort(int[] nums, int left, int right, int index) {
            int pivotindex = partition(nums, left, right);
            if (pivotindex == index) {
                return nums[pivotindex];
            } else {
                return pivotindex < index ? quickSort(nums, pivotindex + 1, right, index) : quickSort(nums, left, pivotindex - 1, index);

            }
        }

        private int partition(int[] nums, int left, int right) {
            int pivot = nums[left];
            while (left < right) {
                while (left < right && nums[right] >= pivot) {
                    right--;
                }
                nums[left] = nums[right];
                while (left < right && nums[left] <= pivot) {
                    left++;
                }
                nums[right] = nums[left];
            }
            nums[left] = pivot;
            return left;
        }

    }
//leetcode submit region end(Prohibit modification and deletion)

}